/*******************************************************************************
 * Copyright (c) 2013-1-31 @author <a href="mailto:iffiff1@hotmail.com">Tyler Chen</a>.
 * All rights reserved.
 *
 * Contributors:
 *     <a href="mailto:iffiff1@hotmail.com">Tyler Chen</a> - initial API and implementation
 ******************************************************************************/
package com.foreveross.bsl.common.utils.bean;

import java.io.*;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Properties;
import java.util.ResourceBundle;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.mail.*;
import javax.mail.internet.*;
import javax.mail.internet.MimeMessage.RecipientType;

/**
 * @author <a href="mailto:2416356969@qq.com">Lee</a>
 * @since 2013-6-24
 */
public class MailHelper {

	private static Logger logger = LoggerFactory.getLogger(MailHelper.class);
	ResourceBundle rb = ResourceBundle.getBundle("config");
	private MimeMessage mimeMessage = null;
	private String saveAttachPath = ""; // 附件下载后的存放目录
	private StringBuffer bodytext = new StringBuffer();// 存放邮件内容
	private String dateformat = "yy-MM-dd HH:mm"; // 默认的日前显示格式

	public MailHelper() {
	}

	public MailHelper(MimeMessage mimeMessage) {
		this.mimeMessage = mimeMessage;
	}

	private static String generateActivateLink(String accountUserName) {
		String randomCode = new Date().toString();
		return md5(accountUserName + ":" + randomCode);
	}

	private static String md5(String string) {
		MessageDigest md = null;
		try {
			md = MessageDigest.getInstance("md5");
			md.update(string.getBytes());
			byte[] md5Bytes = md.digest();
			return bytes2Hex(md5Bytes);
		} catch (NoSuchAlgorithmException e) {
			logger.error(e.getMessage());
			throw new RuntimeException(e);
		}
	}

	public static String getMD5String(String string) {

		MessageDigest md = null;
		try {
			md = MessageDigest.getInstance("md5");
			md.update(string.getBytes());
			byte[] md5Bytes = md.digest();
			return bytes2Hex(md5Bytes);
		} catch (NoSuchAlgorithmException e) {
			logger.error(e.getMessage());
			throw new RuntimeException(e);
		}
	}

	private static String bytes2Hex(byte[] byteArray) {
		StringBuffer strBuf = new StringBuffer();
		for (int i = 0; i < byteArray.length; i++) {
			if (byteArray[i] >= 0 && byteArray[i] < 16) {
				strBuf.append("0");
			}
			strBuf.append(Integer.toHexString(byteArray[i] & 0xFF));
		}
		return strBuf.toString();
	}

	public void setMimeMessage(MimeMessage mimeMessage) {
		this.mimeMessage = mimeMessage;
	}

	public String getCheckCode(String accountUserName) {
		String checkCode = "";
		checkCode = MailHelper.generateActivateLink(accountUserName);
		return checkCode;
	}

	public String getEmailContent(String accountUserName, String checkCode) {
		// String checkCode = "";
		// checkCode = MailHelper.generateActivateLink(accountUserName);
		String phone = rb.getString("tel");
		String activeUrl = "#activateUrl#" + "?" + "checkcode=";
		String content = "尊敬的用户：<br/>"
				+ "&nbsp;&nbsp;&nbsp;您好，感谢您使用变色龙平台，请您在24小时内点击<a href='"
				+ activeUrl + checkCode
				+ "'>激活帐户</a>以激活使用您的帐户。如有问题请致电我们的客服.<br/>" + "联系电话：" + phone;
		return content;
	}

	/**
	 * @param email
	 * @param accountUserName
	 * @return
	 */
	public boolean sendValidateMail(String email, String content) {

		String from = "", to = "", username = "", password = "", smtpServer = "", activateUrl = "";
		from = rb.getString("from");
		// to = rb.getString("to");
		username = rb.getString("username");
		password = rb.getString("password");
		smtpServer = rb.getString("smtpServer");
		activateUrl = rb.getString("activateUrl");

		to = email;
		content = content.replaceAll("#activateUrl#", activateUrl);

		boolean b = true;
		try {
			// 创建 properties ，里面包含了发送邮件服务器的地址。
			Properties mailProps = new Properties();
			mailProps.put("mail.smtp.host", smtpServer);
			mailProps.put("mail.smtp.auth", "true");

			// 创建 session
			Session mailSession = Session.getDefaultInstance(mailProps,
					new SmtpAuthenticator(username, password));

			// 创建 邮件的message，message对象包含了邮件众多有的部件，都是封装成了set方法去设置的
			MimeMessage message = new MimeMessage(mailSession);

			try {
				message.setSubject("变色龙帐户激活邮件");
				message.setSentDate(new Date());
				message.setFrom(new InternetAddress(from));
				message.setRecipient(RecipientType.TO, new InternetAddress(to));

				message.setContent(content, "text/html;charset=utf-8");
				// 发送邮件
				Transport.send(message);
			} catch (Exception e) {
				logger.error(e.getMessage());
				b = false;
			}
		} catch (Exception e) {
			logger.error(e.getMessage());
			b = false;
		}

		return b;
	}

	public String sendMain(String emailAddress, String userName) {

		String senderAddress = "2416356969@qq.com";
		String reciverAddress = emailAddress;

		String username = "2416356969";
		String password = "*******";

		String sendMailResult = "";

		try {
			// 创建 properties ，里面包含了发送邮件服务器的地址。
			Properties mailProps = new Properties();
			mailProps.put("mail.smtp.host", "smtp.qq.com");
			mailProps.put("mail.smtp.auth", "true");

			// 创建 session
			Session mailSession = Session.getDefaultInstance(mailProps,
					new SmtpAuthenticator(username, password));

			// 创建 邮件的message，message对象包含了邮件众多有的部件，都是封装成了set方法去设置的
			MimeMessage message = new MimeMessage(mailSession);

			// 设置发信人
			message.setFrom(new InternetAddress(senderAddress));
			// 收信人
			message.setRecipient(Message.RecipientType.TO, new InternetAddress(
					reciverAddress));

			// 邮件标题
			message.setSubject("a poem");
			message.setSentDate(new Date());// 发送日期

			// 创建附件
			MimeMultipart multi = new MimeMultipart("related");

			// 创建 BodyPart，主要作用是将以后创建的n个内容加入MimeMultipart.也就是可以发n个附件

			/** *************************************************** */

			BodyPart textBodyPart = new MimeBodyPart(); // 第一个BodyPart.主要写一些一般的信件内容。

			textBodyPart
					.setText("Do you fear the force of the wind The slash of the rain Go face them and fight them "
							+ "Be savage again. Go hungry and cold like the wolf, Go wade like the crane: "
							+ "The palms of your hands will thicken, The skin of your cheek will tan, You'll grow ragged and weary and swarthy, "
							+ "But you'll walk like a man!");

			// 压入第一个BodyPart到MimeMultipart对象中。
			multi.addBodyPart(textBodyPart);

			/** *************************************************** */

			// 创建第二个BodyPart,是一个FileDataSource
			// FileDataSource fds = new FileDataSource("e:/注册码.txt");
			//
			// BodyPart fileBodyPart = new MimeBodyPart(); //第二个BodyPart
			// fileBodyPart.setDataHandler(new DataHandler(fds)); //字符流形式装入文件
			// fileBodyPart.setFileName(fds.getName()); //设置文件名，可以不是原来的文件名。
			//
			// multi.addBodyPart(fileBodyPart);

			// MimeMultPart作为Content加入message
			message.setContent(multi);

			// 所有以上的工作必须保存。
			message.saveChanges();

			Transport.send(message);
			System.out.println("Mail send succesfully!");

			sendMailResult = "Mail send succesfully!";
		} catch (Exception e) {
			e.printStackTrace();
			sendMailResult = "Mail send fail!";
		}

		return sendMailResult;
	}

	/**
	 * 获得发件人的地址和姓名
	 */
	public String getFrom() throws Exception {
		InternetAddress address[] = (InternetAddress[]) mimeMessage.getFrom();
		String from = address[0].getAddress();
		if (from == null)
			from = "";
		String personal = address[0].getPersonal();
		if (personal == null)
			personal = "";
		String fromaddr = personal + "<" + from + ">";
		return fromaddr;
	}

	/**
	 * 获得邮件的收件人，抄送，和密送的地址和姓名，根据所传递的参数的不同 "to"----收件人 "cc"---抄送人地址 "bcc"---密送人地址
	 */
	public String getMailAddress(String type) throws Exception {
		String mailaddr = "";
		String addtype = type.toUpperCase();
		InternetAddress[] address = null;
		if (addtype.equals("TO") || addtype.equals("CC")
				|| addtype.equals("BCC")) {
			if (addtype.equals("TO")) {
				address = (InternetAddress[]) mimeMessage
						.getRecipients(Message.RecipientType.TO);
			} else if (addtype.equals("CC")) {
				address = (InternetAddress[]) mimeMessage
						.getRecipients(Message.RecipientType.CC);
			} else {
				address = (InternetAddress[]) mimeMessage
						.getRecipients(Message.RecipientType.BCC);
			}
			if (address != null) {
				for (int i = 0; i < address.length; i++) {
					String email = address[i].getAddress();
					if (email == null)
						email = "";
					else {
						email = MimeUtility.decodeText(email);
					}
					String personal = address[i].getPersonal();
					if (personal == null)
						personal = "";
					else {
						personal = MimeUtility.decodeText(personal);
					}
					String compositeto = personal + "<" + email + ">";
					mailaddr += "," + compositeto;
				}
				mailaddr = mailaddr.substring(1);
			}
		} else {
			throw new Exception("Error emailaddr type!");
		}
		return mailaddr;
	}

	/**
	 * 获得邮件主题
	 */
	public String getSubject() throws MessagingException {
		String subject = "";
		try {
			subject = MimeUtility.decodeText(mimeMessage.getSubject());
			if (subject == null)
				subject = "";
		} catch (Exception exce) {
		}
		return subject;
	}

	/**
	 * 获得邮件发送日期
	 */
	public String getSentDate() throws Exception {
		Date sentdate = mimeMessage.getSentDate();
		SimpleDateFormat format = new SimpleDateFormat(dateformat);
		return format.format(sentdate);
	}

	/**
	 * 获得邮件正文内容
	 */
	public String getBodyText() {
		return bodytext.toString();
	}

	/**
	 * 解析邮件，把得到的邮件内容保存到一个StringBuffer对象中，解析邮件 主要是根据MimeType类型的不同执行不同的操作，一步一步的解析
	 */
	public void getMailContent(Part part) throws Exception {
		String contenttype = part.getContentType();
		int nameindex = contenttype.indexOf("name");
		boolean conname = false;
		if (nameindex != -1)
			conname = true;
		System.out.println("CONTENTTYPE: " + contenttype);
		if (part.isMimeType("text/plain") && !conname) {
			bodytext.append((String) part.getContent());
		} else if (part.isMimeType("text/html") && !conname) {
			bodytext.append((String) part.getContent());
		} else if (part.isMimeType("multipart/*")) {
			Multipart multipart = (Multipart) part.getContent();
			int counts = multipart.getCount();
			for (int i = 0; i < counts; i++) {
				getMailContent(multipart.getBodyPart(i));
			}
		} else if (part.isMimeType("message/rfc822")) {
			getMailContent((Part) part.getContent());
		} else {
		}
	}

	/**
	 * 判断此邮件是否需要回执，如果需要回执返回"true",否则返回"false"
	 */
	public boolean getReplySign() throws MessagingException {
		boolean replysign = false;
		String needreply[] = mimeMessage
				.getHeader("Disposition-Notification-To");
		if (needreply != null) {
			replysign = true;
		}
		return replysign;
	}

	/**
	 * 获得此邮件的Message-ID
	 */
	public String getMessageId() throws MessagingException {
		return mimeMessage.getMessageID();
	}

	/**
	 * 【判断此邮件是否已读，如果未读返回返回false,反之返回true】
	 */
	public boolean isNew() throws MessagingException {
		boolean isnew = false;
		Flags flags = ((Message) mimeMessage).getFlags();
		Flags.Flag[] flag = flags.getSystemFlags();
		System.out.println("flags's length: " + flag.length);
		for (int i = 0; i < flag.length; i++) {
			if (flag[i] == Flags.Flag.SEEN) {
				isnew = true;
				System.out.println("seen Message.......");
				break;
			}
		}
		return isnew;
	}

	/**
	 * 判断此邮件是否包含附件
	 */
	public boolean isContainAttach(Part part) throws Exception {
		boolean attachflag = false;
		// String contentType = part.getContentType();
		if (part.isMimeType("multipart/*")) {
			Multipart mp = (Multipart) part.getContent();
			for (int i = 0; i < mp.getCount(); i++) {
				BodyPart mpart = mp.getBodyPart(i);
				String disposition = mpart.getDisposition();
				if ((disposition != null)
						&& ((disposition.equals(Part.ATTACHMENT)) || (disposition
								.equals(Part.INLINE))))
					attachflag = true;
				else if (mpart.isMimeType("multipart/*")) {
					attachflag = isContainAttach((Part) mpart);
				} else {
					String contype = mpart.getContentType();
					if (contype.toLowerCase().indexOf("application") != -1)
						attachflag = true;
					if (contype.toLowerCase().indexOf("name") != -1)
						attachflag = true;
				}
			}
		} else if (part.isMimeType("message/rfc822")) {
			attachflag = isContainAttach((Part) part.getContent());
		}
		return attachflag;
	}

	/**
	 * 【保存附件】
	 */
	public void saveAttachMent(Part part) throws Exception {
		String fileName = "";
		if (part.isMimeType("multipart/*")) {
			Multipart mp = (Multipart) part.getContent();
			for (int i = 0; i < mp.getCount(); i++) {
				BodyPart mpart = mp.getBodyPart(i);
				String disposition = mpart.getDisposition();
				if ((disposition != null)
						&& ((disposition.equals(Part.ATTACHMENT)) || (disposition
								.equals(Part.INLINE)))) {
					fileName = mpart.getFileName();
					if (fileName.toLowerCase().indexOf("gb2312") != -1) {
						fileName = MimeUtility.decodeText(fileName);
					}
					saveFile(fileName, mpart.getInputStream());
				} else if (mpart.isMimeType("multipart/*")) {
					saveAttachMent(mpart);
				} else {
					fileName = mpart.getFileName();
					if ((fileName != null)
							&& (fileName.toLowerCase().indexOf("GB2312") != -1)) {
						fileName = MimeUtility.decodeText(fileName);
						saveFile(fileName, mpart.getInputStream());
					}
				}
			}
		} else if (part.isMimeType("message/rfc822")) {
			saveAttachMent((Part) part.getContent());
		}
	}

	/**
	 * 【设置附件存放路径】
	 */

	public void setAttachPath(String attachpath) {
		this.saveAttachPath = attachpath;
	}

	/**
	 * 【设置日期显示格式】
	 */
	public void setDateFormat(String format) throws Exception {
		this.dateformat = format;
	}

	/**
	 * 【获得附件存放路径】
	 */
	public String getAttachPath() {
		return saveAttachPath;
	}

	/**
	 * 【真正的保存附件到指定目录里】
	 */
	private void saveFile(String fileName, InputStream in) throws Exception {
		String osName = System.getProperty("os.name");
		String storedir = getAttachPath();
		String separator = "";
		if (osName == null)
			osName = "";
		if (osName.toLowerCase().indexOf("win") != -1) {
			separator = "\\";
			if (storedir == null || storedir.equals(""))
				storedir = "c:\\tmp";
		} else {
			separator = "/";
			storedir = "/tmp";
		}
		File storefile = new File(storedir + separator + fileName);
		System.out.println("storefile's path: " + storefile.toString());
		// for(int i=0;storefile.exists();i++){
		// storefile = new File(storedir+separator+fileName+i);
		// }
		BufferedOutputStream bos = null;
		BufferedInputStream bis = null;
		try {
			bos = new BufferedOutputStream(new FileOutputStream(storefile));
			bis = new BufferedInputStream(in);
			int c;
			while ((c = bis.read()) != -1) {
				bos.write(c);
				bos.flush();
			}
		} catch (Exception exception) {
			exception.printStackTrace();
			throw new Exception("文件保存失败!");
		} finally {
			bos.close();
			bis.close();
		}
	}

	/**
	 * PraseMimeMessage类测试
	 */
	public void ReceiveMail() throws Exception {
		Properties props = System.getProperties();
		props.put("mail.smtp.host", "smtp.163.com");
		props.put("mail.smtp.auth", "true");
		Session session = Session.getDefaultInstance(props, null);
		URLName urln = new URLName("pop3", "pop3.163.com", 110, null,
				"xiangzhengyan", "pass");
		Store store = session.getStore(urln);
		store.connect();
		Folder folder = store.getFolder("INBOX");
		folder.open(Folder.READ_ONLY);
		Message message[] = folder.getMessages();
		System.out.println("Messages's length: " + message.length);
		MailHelper pmm = null;
		for (int i = 0; i < message.length; i++) {
			System.out.println("======================");
			pmm = new MailHelper((MimeMessage) message[i]);
			System.out
					.println("Message " + i + " subject: " + pmm.getSubject());
			System.out.println("Message " + i + " sentdate: "
					+ pmm.getSentDate());
			System.out.println("Message " + i + " replysign: "
					+ pmm.getReplySign());
			System.out.println("Message " + i + " hasRead: " + pmm.isNew());
			System.out.println("Message " + i + "  containAttachment: "
					+ pmm.isContainAttach((Part) message[i]));
			System.out.println("Message " + i + " form: " + pmm.getFrom());
			System.out.println("Message " + i + " to: "
					+ pmm.getMailAddress("to"));
			System.out.println("Message " + i + " cc: "
					+ pmm.getMailAddress("cc"));
			System.out.println("Message " + i + " bcc: "
					+ pmm.getMailAddress("bcc"));
			pmm.setDateFormat("yy年MM月dd日 HH:mm");
			System.out.println("Message " + i + " sentdate: "
					+ pmm.getSentDate());
			System.out.println("Message " + i + " Message-ID: "
					+ pmm.getMessageId());
			// 获得邮件内容===============
			pmm.getMailContent((Part) message[i]);
			System.out.println("Message " + i + " bodycontent: \r\n"
					+ pmm.getBodyText());
			pmm.setAttachPath("c:\\");
			pmm.saveAttachMent((Part) message[i]);
		}
	}

	class SmtpAuthenticator extends Authenticator {
		String username = null;
		String password = null;

		// SMTP身份验证
		public SmtpAuthenticator(String username, String password) {
			this.username = username;
			this.password = password;
		}

		public PasswordAuthentication getPasswordAuthentication() {
			return new PasswordAuthentication(this.username, this.password);
		}
	}

	public static void main(String[] args) {
		MailHelper m = new MailHelper();
		String c = m.getCheckCode("a");
		String con = m.getEmailContent("a", c);
		con = con.replace("#activeUrl#", "test");
		System.out.println(con);
	}

}
